package com.briup.mock.result;

import lombok.AllArgsConstructor;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;

import java.util.List;
import java.util.Set;

/**
 * 初始化训练需要的参数
 */
@Getter
@Setter
@NoArgsConstructor
@AllArgsConstructor
public class ZookeeperInitResult extends ZookeeperResult {
    //模型的最后一个版本,没有版本就传vo
    private String lastVersion;
    //分辨率
    private String reslution;
    //迭代次数
    private String iterateTimes;
    //算子类型
    private Integer operatorType;
    //网络结构
    private String networkStructure;
    //优化器
    private String optimizer;
    //损失函数
    private String lossValue;
    //模型文件保存地址
    private String newModelPath;
    //训练记录id
//    private Integer trainingId;
    // 分类指标
//    private List<TrainingLabelVo> trainingLabelVoList;
    //训练所用数据集id
    private List<Integer> datasetIdList;

    // 训练信息
//    private Training training;
    // 分类信息
    private Set<String> labelNameSet;
}
